from django.db.models import QuerySet, Manager


class GroupQuerySet(QuerySet):
    """
    这个管理器增加了根据 values 或 values_list 分组的功能
    """

    def group_by(self):
        clone = self._chain()
        clone.query.group_by = True
        return clone


class GroupManager(Manager):
    _queryset_class = GroupQuerySet
